# 自定义filter的基本语法
from django import template
from login.models import UserInfo
from index.models import InformationFormData

''' 
# 将自定义的 filter 注册到程序中
@register.filter
@register.filter(name = 'fleter_name') 这个装饰器把自定义的filter注册到程序中，
让其可以使用自定义的 filter 方法； reg是 ‘|’ 前面的变量 reg2是方法后面的参数。
'''

register = template.Library() # 实例化一个模板对象

@register.filter(name = 'getUserFormState')
def getUserFormState(userState, userName):
    if userState == 0:
        return False
    else:
        state = None
        result = InformationFormData.objects.filter(form_76= userName).last()
        if result.form_78 == 0:
            state = '待审核'
        else:
            state = '已审核'
        return state